''' 
NavSquare Program with functions - final
Mission 3 Objectives 10-11
'''
from botcore import *
from time import sleep

motors.enable(True)
delay = 2.0

def go_straight(speed, delay):
    motors.run(LEFT, speed)
    motors.run(RIGHT, speed)
    sleep(delay)

def turn_90(speed, delay):
    motors.run(LEFT, speed)
    motors.run(RIGHT, -speed)
    sleep(delay)

# countdown
def countdown():
    leds.user_num(4, True)
    sleep(1.0)
    leds.user_num(3, True)
    sleep(1.0)

# -- Main Program --
countdown()

if buttons.was_pressed(0):
    leds.user_num(0, True)
    go_straight(30, delay)
    turn_90(20, 0.72)
    go_straight(30, delay)
    turn_90(20, 0.72)
    go_straight(30, delay)
    turn_90(20, 0.72)
    go_straight(30, delay)
    turn_90(20, 0.72)

elif buttons.was_pressed(1):
    leds.user_num(7, True)
    go_straight(30, delay)
    turn_90(-20, 0.72)
    go_straight(30, delay)
    turn_90(-20, 0.72)
    go_straight(30, delay)
    turn_90(-20, 0.72)
    go_straight(30, delay)
    turn_90(-20, 0.72)

else:
    leds.user(0)
    motors.run(LEFT, 0)
    motors.run(RIGHT, 0)
    motors.enable(False)